//一位老师正在出一场由 n 道判断题构成的考试，每道题的答案为 true （用 'T' 表示）或者 false （用 'F' 表示）。老师想增加学生对自己做出
//答案的不确定性，方法是 最大化 有 连续相同 结果的题数。（也就是连续出现 true 或者连续出现 false）。 
//
// 给你一个字符串 answerKey ，其中 answerKey[i] 是第 i 个问题的正确结果。除此以外，还给你一个整数 k ，表示你能进行以下操作的最
//多次数： 
//
// 
// 每次操作中，将问题的正确答案改为 'T' 或者 'F' （也就是将 answerKey[i] 改为 'T' 或者 'F' ）。 
// 
//
// 请你返回在不超过 k 次操作的情况下，最大 连续 'T' 或者 'F' 的数目。 
//
// 
//
// 示例 1： 
//
// 
//输入：answerKey = "TTFF", k = 2
//输出：4
//解释：我们可以将两个 'F' 都变为 'T' ，得到 answerKey = "TTTT" 。
//总共有四个连续的 'T' 。
// 
//
// 示例 2： 
//
// 
//输入：answerKey = "TFFT", k = 1
//输出：3
//解释：我们可以将最前面的 'T' 换成 'F' ，得到 answerKey = "FFFT" 。
//或者，我们可以将第二个 'T' 换成 'F' ，得到 answerKey = "TFFF" 。
//两种情况下，都有三个连续的 'F' 。
// 
//
// 示例 3： 
//
// 
//输入：answerKey = "TTFTTFTT", k = 1
//输出：5
//解释：我们可以将第一个 'F' 换成 'T' ，得到 answerKey = "TTTTTFTT" 。
//或者我们可以将第二个 'F' 换成 'T' ，得到 answerKey = "TTFTTTTT" 。
//两种情况下，都有五个连续的 'T' 。
// 
//
// 
//
// 提示： 
//
// 
// n == answerKey.length 
// 1 <= n <= 5 * 104 
// answerKey[i] 要么是 'T' ，要么是 'F' 
// 1 <= k <= n 
// 
// Related Topics 字符串 二分查找 前缀和 滑动窗口 
// 👍 133 👎 0

package com.cute.leetcode.editor.cn;

import java.util.ArrayList;
import java.util.List;

public class MaximizeTheConfusionOfAnExam {
    public static void main(String[] args) {
        new MaximizeTheConfusionOfAnExam().new Solution().maxConsecutiveAnswers("FFFTTFTTFT", 3);
    }
    //leetcode submit region begin(Prohibit modification and deletion)
    class Solution {
        String s;
        int n, k;
        /**
         * 这一题和第1004题是一样的，只是TF需要分别考虑了
         * 以全改为T为例，此时窗口中最多只能有k个F，超过K个时就需要移动左边界直到移出一个F
         */
        public int maxConsecutiveAnswers(String answerKey, int k) {
            s = answerKey;
            n = s.length();
            this.k = k;
            return Math.max(getLen('T'), getLen('F'));
        }

        private int getLen(char c){// 需要转换的字符为c
            int ans = 0;
            for (int i = 0, j = 0, cnt = 0; i < n; i++) {
                if (s.charAt(i) == c) cnt++;// 出现了字符c的话计数++
                while (cnt > k){// 如果计数 > k，则必须删除一个字符c
                    if (s.charAt(j) == c) cnt--;
                    j++;
                }
                ans = Math.max(ans, i - j + 1);
            }
            return ans;
        }

        // 我自己的写法，也是查两个字符
        public int myMaxConsecutiveAnswers(String answerKey, int k) {
            int max = 0;
            int left = 0;
            int right = 0;
            char c = 'T';
            int n = k;
            while (right < answerKey.length()){
                while (right < answerKey.length() && answerKey.charAt(right) == c) right++;
                if (right < answerKey.length()){
                    if (k > 0) k--;
                    else {
                        max = Math.max(max, right - left);
                        while (left < answerKey.length() && answerKey.charAt(left) == c) left++;
                        left++;
                    }
                    right++;
                }
            }
            max = Math.max(max, right - left);
            c = 'F';
            left = 0;
            right = 0;
            k = n;
            while (right < answerKey.length()){
                while (right < answerKey.length() && answerKey.charAt(right) == c) right++;
                if (right < answerKey.length()){
                    if (k > 0) k--;
                    else {
                        max = Math.max(max, right - left);
                        while (left < answerKey.length() && answerKey.charAt(left) == c) left++;
                        left++;
                    }
                    right++;
                }
            }
            return Math.max(max, right - left);
        }
    }
//leetcode submit region end(Prohibit modification and deletion)

}